from torchsummary import summary

from nets.pspnet import PSPNet

model = PSPNet(num_classes=21, backbone="mobilenet", downsample_factor=16, aux_branch=False,
               pretrained=False).train().cuda()

summary(model, (3, 473, 473))
